
# tools/base64_tool.py
import base64

def base64_encode(data: str) -> str:
    return base64.b64encode(data.encode()).decode()

def base64_decode(encoded_data: str) -> str:
    return base64.b64decode(encoded_data).decode()

def run():
    print("\n欢迎使用 Base64 编解码工具")
    print("1. 编码")
    print("2. 解码")
    choice = input("请选择操作（1 或 2）：")
    if choice == "1":
        data = input("请输入要编码的字符串：")
        result = base64_encode(data)
        print(f"编码结果：{result}")
    elif choice == "2":
        encoded_data = input("请输入要解码的 Base64 字符串：")
        try:
            result = base64_decode(encoded_data)
            print(f"解码结果：{result}")
        except Exception as e:
            print(f"解码失败：{e}")
    else:
        print("无效选项！")
